you can claim disability benefits for early rehab